home *** CD-ROM | disk | FTP | other *** search
/ PC World 2006 February / PCWorld_2006-02_cd.bin / software / vyzkuste / triky / triky.exe / httrack-3.33.exe / {app} / src / htsthread.c < prev    next >
C/C++ Source or Header  |  2005-02-05  |  5KB  |  220 lines
  1. /* ------------------------------------------------------------ */
  2. /*
  3. HTTrack Website Copier, Offline Browser for Windows and Unix
  4. Copyright (C) Xavier Roche and other contributors
  5.  
  6. This program is free software; you can redistribute it and/or
  7. modify it under the terms of the GNU General Public License
  8. as published by the Free Software Foundation; either version 2
  9. of the License, or any later version.
  10.  
  11. This program is distributed in the hope that it will be useful,
  12. but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. GNU General Public License for more details.
  15.  
  16. You should have received a copy of the GNU General Public License
  17. along with this program; if not, write to the Free Software
  18. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19.  
  20.  
  21. Important notes:
  22.  
  23. - We hereby ask people using this source NOT to use it in purpose of grabbing
  24. emails addresses, or collecting any other private information on persons.
  25. This would disgrace our work, and spoil the many hours we spent on it.
  26.  
  27.  
  28. Please visit our Website: http://www.httrack.com
  29. */
  30.  
  31.  
  32. /* ------------------------------------------------------------ */
  33. /* File: Threads                                                */
  34. /* Author: Xavier Roche                                         */
  35. /* ------------------------------------------------------------ */
  36.  
  37. /* Internal engine bytecode */
  38. #define HTS_INTERNAL_BYTECODE
  39.  
  40. #include "htsglobal.h"
  41. #include "htsbase.h"
  42. #include "htsthread.h"
  43.  
  44. #if USE_BEGINTHREAD
  45. #if HTS_WIN
  46. #include <process.h>
  47. #endif
  48. #endif
  49.  
  50. static int process_chain = 0;
  51. static PTHREAD_LOCK_TYPE process_chain_mutex;
  52.  
  53. HTSEXT_API void htsthread_wait(void ) {
  54.   htsthread_wait_n(0);
  55. }
  56.  
  57. HTSEXT_API void htsthread_wait_n(int n_wait) {
  58. #if USE_BEGINTHREAD
  59.   int wait = 0;
  60.   do {
  61.     htsSetLock(&process_chain_mutex, 1);
  62.     wait = (process_chain > n_wait );
  63.     htsSetLock(&process_chain_mutex, 0);
  64.     if (wait)
  65.       Sleep(100);
  66.   } while(wait);
  67. #endif
  68. }
  69.  
  70. HTSEXT_API void htsthread_init(void ) {
  71. #if USE_BEGINTHREAD
  72.   assertf(process_chain == 0);
  73.   htsSetLock(&process_chain_mutex, -999);
  74. #endif
  75. }
  76.  
  77. HTSEXT_API void htsthread_uninit(void ) {
  78.   htsthread_wait();
  79. #if USE_BEGINTHREAD
  80.   htsSetLock(&process_chain_mutex, -998);
  81. #endif
  82. }
  83.  
  84. typedef struct {
  85.   PTHREAD_TYPE ( PTHREAD_TYPE_FNC *start_address )( void * );
  86.   void** arglist;
  87. } execth_args;
  88. static PTHREAD_TYPE PTHREAD_TYPE_FNC execth( void * arg )
  89. {
  90.   execth_args* args = (execth_args*) arg;
  91.   assertf(args != NULL);
  92.  
  93.   htsSetLock(&process_chain_mutex, 1);
  94.   process_chain++;
  95.   assertf(process_chain > 0);
  96.   htsSetLock(&process_chain_mutex, 0);
  97.  
  98.   (void) args->start_address(args->arglist);
  99.  
  100.   htsSetLock(&process_chain_mutex, 1);
  101.   process_chain--;
  102.   assertf(process_chain >= 0);
  103.   htsSetLock(&process_chain_mutex, 0);
  104.  
  105.   free(arg);
  106.   return PTHREAD_RETURN;
  107. }
  108.  
  109.  
  110. HTSEXT_API int hts_newthread( PTHREAD_TYPE ( PTHREAD_TYPE_FNC *start_address )( void * ), unsigned stack_size, void *arglist )
  111. {
  112.   execth_args* args = (execth_args*) malloc(sizeof(execth_args));
  113.   assertf(args != NULL);
  114.   args->start_address = start_address;
  115.   args->arglist = arglist;
  116.  
  117.   /* create a thread */
  118. #ifdef _WIN32
  119.   if (_beginthread(execth, stack_size, args) == -1) {
  120.     free(args);
  121.     return -1;
  122.   }
  123. #else
  124.   {
  125.     PTHREAD_HANDLE handle = 0;
  126.     int retcode;
  127.     retcode = pthread_create(&handle, NULL, execth, args);
  128.     if (retcode != 0) {   /* error */
  129.       free(args);
  130.       return -1;
  131.     } else {
  132.       /* detach the thread from the main process so that is can be independent */
  133.       pthread_detach(handle);
  134.     }
  135.   }
  136. #endif
  137.   return 0;
  138. }
  139.  
  140.  
  141. // Threads - emulate _beginthread under Linux/Unix using pthread_XX
  142. // Some changes will have to be done, see PTHREAD_RETURN,PTHREAD_TYPE
  143. #if USE_PTHREAD
  144. #include <pthread.h>    /* _beginthread, _endthread */
  145. #if 0
  146. unsigned long _beginthread( void* ( *start_address )( void * ), unsigned stack_size, void *arglist )
  147. {
  148.   pthread_t th;
  149.   int retcode;
  150.   /* create a thread */
  151.   retcode = pthread_create(&th, NULL, start_address, arglist);
  152.   if (retcode != 0)     /* error */
  153.     return -1;
  154.   /* detach the thread from the main process so that is can be independent */
  155.   pthread_detach(th);
  156.   return 0;
  157. }
  158. #endif
  159. #endif
  160.  
  161. #if USE_BEGINTHREAD
  162. /* 
  163.   Simple lock function
  164.  
  165.   Return value: always 0
  166.   Parameter:
  167.   1 wait for lock (mutex) available and lock it
  168.   0 unlock the mutex
  169.   [-1 check if locked (always return 0 with mutex)]
  170.   -999 initialize
  171.   -998 free
  172.   */
  173. HTSEXT_API int htsSetLock(PTHREAD_LOCK_TYPE* hMutex,int lock) {
  174. #if HTS_WIN
  175.   /* lock */
  176.   switch(lock) {
  177.    case 1:    /* lock */
  178.      assertf(*hMutex != NULL);
  179.      WaitForSingleObject(*hMutex,INFINITE);
  180.      break;
  181.    case 0:     /* unlock */
  182.      assertf(*hMutex != NULL);
  183.      ReleaseMutex(*hMutex);
  184.      break;
  185.    case -999:  /* create */
  186.      *hMutex=CreateMutex(NULL,FALSE,NULL);
  187.      break;
  188.    case -998:  /* destroy */
  189.      CloseHandle(*hMutex);
  190.      *hMutex = NULL;
  191.      break;
  192.    default:
  193.      assert(FALSE);
  194.      break;
  195.   }
  196. #else
  197.   switch(lock) {
  198.    case 1:    /* lock */
  199.      pthread_mutex_lock(hMutex);
  200.      break;
  201.    case 0:     /* unlock */
  202.      pthread_mutex_unlock(hMutex);
  203.      break;
  204.    case -999:  /* create */
  205.      pthread_mutex_init(hMutex,0);
  206.      break;
  207.    case -998:  /* destroy */
  208.      pthread_mutex_destroy(hMutex);
  209.      break;
  210.    default:
  211.      assert(0);
  212.      break;
  213.   }
  214. #endif
  215.   return 0;
  216. }
  217.  
  218. #endif
  219.  
  220.